﻿/********************************************************
 *  ██████╗  ██████╗████████╗██╗
 * ██╔════╝ ██╔════╝╚══██╔══╝██║
 * ██║  ███╗██║        ██║   ██║
 * ██║   ██║██║        ██║   ██║
 * ╚██████╔╝╚██████╗   ██║   ███████╗
 *  ╚═════╝  ╚═════╝   ╚═╝   ╚══════╝
 * Geophysical Computational Tools & Library (GCTL)
 *
 * Copyright (c) 2023  Yi Zhang (yizhang-geo@zju.edu.cn)
 *
 * GCTL is distributed under a dual licensing scheme. You can redistribute 
 * it and/or modify it under the terms of the GNU Lesser General Public 
 * License as published by the Free Software Foundation, either version 2 
 * of the License, or (at your option) any later version. You should have 
 * received a copy of the GNU Lesser General Public License along with this 
 * program. If not, see <http://www.gnu.org/licenses/>.
 * 
 * If the terms and conditions of the LGPL v.2. would prevent you from using 
 * the GCTL, please consider the option to obtain a commercial license for a 
 * fee. These licenses are offered by the GCTL's original author. As a rule, 
 * licenses are provided "as-is", unlimited in time for a one time fee. Please 
 * send corresponding requests to: yizhang-geo@zju.edu.cn. Please do not forget 
 * to include some description of your company and the realm of its activities. 
 * Also add information on how to contact you by electronic and paper mail.
 ******************************************************/

#ifndef _GCTL_REGULAR_GRID_H
#define _GCTL_REGULAR_GRID_H

#include "mesh.h"
#include "../geometry/rectangle2d.h"
#include "../io/gmsh_io.h"
#include "../io/surfer_io.h"

#ifdef GCTL_NETCDF
#include "../io/netcdf_io.h"
#endif // GCTL_NETCDF

namespace gctl
{
	class regular_grid : public base_mesh
	{
	public:

		/**
		 * 以下为mesh类型的虚函数实现
		 */

		void init(std::string in_name, std::string in_info, int xnum, int ynum, 
			double xmin, double ymin, double dx, double dy);

		void load_binary(std::string filename);
		void save_binary(std::string filename);

		void *get_node(unsigned int idx);
		void *get_element(unsigned int idx);

		/**
		 * 以下为regular_grid的专有函数
		 */

		regular_grid();
		regular_grid(std::string in_name, std::string in_info, int xnum, int ynum, 
			double xmin, double ymin, double dx, double dy);
		virtual ~regular_grid();

		int get_xdim() const;
		int get_ydim() const;
		double get_xmin() const;
		double get_ymin() const;
		double get_dx() const;
		double get_dy() const;

#ifdef GCTL_NETCDF

		void load_netcdf_grid(std::string filename, mesh_data_type_e d_type, std::string xname = "x", 
			std::string yname = "y", std::string dataname = "z");
		void save_netcdf_grid(std::string filename, mesh_data_type_e d_type);
		void save_netcdf_grid(std::string filename, std::string datname);

#endif // GCTL_NETCDF

		void load_surfer_grid(std::string filename, std::string datname, mesh_data_type_e d_type, surfer_file_type_e grid_type = Surfer7Grid);
		void save_surfer_grid(std::string filename, std::string datname, surfer_file_type_e grid_type = Surfer7Grid);

		void save_gmsh(std::string filename, index_packed_e packed = Packed);
		void save_gmsh(std::string filename, mesh_data_type_e d_type, output_type_e out_mode, index_packed_e packed = Packed);
		void save_gmsh(std::string filename, std::string datname, output_type_e out_mode, index_packed_e packed = Packed);

		void load_data_cloud(const array<point2dc> &in_posi, const array<double> &in_val, double search_xlen, 
			double search_ylen, double search_deg, std::string datname, mesh_data_type_e d_type = NodeData);

		void extract_points(std::string datname, const array<point2dc> &in_posi, array<double> &out_val);
		void extract_profile(std::string datname, const point2dc &start_p, const point2dc &end_p, int size_p, 
			array<point2dc> &out_posi, array<double> &out_val);

		void gradient(std::string datname, std::string gradname, gradient_type_e d_type, int order = 1);

#ifdef GCTL_WAVELIB
		void wavelet(std::string datname, std::string wavename, int order, bool summary = true);
#endif // GCTL_WAVELIB

	protected:
		int rg_xnum, rg_ynum;
		double rg_xmin, rg_ymin;
		double rg_dx, rg_dy;
		array<vertex2dc> nodes;
		array<rectangle2d> elements;
	};
}

#endif //_GCTL_REGULAR_GRID_H